package com.w.picture.service.impl;

import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.util.ObjUtil;
import cn.hutool.core.util.StrUtil;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.w.picture.exception.BusinessException;
import com.w.picture.exception.ErrorCode;
import com.w.picture.exception.ThrowUtils;
import com.w.picture.manager.CosManager;
import com.w.picture.manager.sharding.DynamicShardingManager;
import com.w.picture.mapper.SpaceMapper;
import com.w.picture.mode.dto.space.SpaceAddRequest;
import com.w.picture.mode.dto.space.SpaceQueryRequest;
import com.w.picture.mode.entity.Space;
import com.w.picture.mode.entity.SpaceUser;
import com.w.picture.mode.entity.User;
import com.w.picture.mode.enums.InviteStatus;
import com.w.picture.mode.enums.SpaceLevelEnum;
import com.w.picture.mode.enums.SpaceRoleEnum;
import com.w.picture.mode.enums.SpaceTypeEnum;
import com.w.picture.mode.vo.SpaceVO;
import com.w.picture.mode.vo.UserVo;
import com.w.picture.service.SpaceService;
import com.w.picture.service.SpaceUserService;
import com.w.picture.service.UserService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.BeanUtils;
import org.springframework.context.annotation.Lazy;
import org.springframework.stereotype.Service;
import org.springframework.transaction.support.TransactionTemplate;

import javax.annotation.Resource;
import javax.servlet.http.HttpServletRequest;
import java.util.List;
import java.util.Map;
import java.util.Optional;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;
import java.util.stream.Collectors;

/**
 * @author wzp
 * @description 针对表【space(图片)】的数据库操作Service实现
 * @createDate 2024-12-31 11:45:13
 */
@Slf4j
@Service
public class SpaceServiceImpl extends ServiceImpl<SpaceMapper, Space> implements SpaceService {

    @Resource
    private UserService userService;
    @Resource
    private CosManager cosManager;
    @Resource
    private TransactionTemplate transactionTemplate;
    @Resource
    @Lazy
    private SpaceUserService spaceUserService;

//    可选操作:由于启动类移除了分库分表配置,因此,代码中也注释掉.
//    @Resource
//    @Lazy
//    private DynamicShardingManager dynamicShardingManager;

    private Map<Long, Object> lockMap = new ConcurrentHashMap<>();

    @Override
    public long addSpace(SpaceAddRequest spaceAddRequest, User lgoinUser) {
        // 1.填充参数默认值
        Space space = new Space();
        BeanUtils.copyProperties(spaceAddRequest, space);
        if (StrUtil.isBlank(space.getSpaceName())) {
            space.setSpaceName("默认空间");
        }
        // 增加补充逻辑:如果空间级别为空,默认为普通级别
        if (space.getSpaceLevel() == null) {
            space.setSpaceLevel(SpaceLevelEnum.COMMON.getValue());
        }
        // 增加补充逻辑:如果空间类型为空,默认为私有空间
        if (space.getSpaceType() == null) {
            space.setSpaceType(SpaceTypeEnum.PRIVATE.getValue());
        }
        // 填充容量和大小
        this.fillSpaceBySpaceLevel(space);

        // 2.校验参数
        this.validSpace(space, true);
        // 3.校验权限,非管理员只能创建普通级别空间
        Long userId = lgoinUser.getId();
        space.setUserId(userId);
//        // 非管理员只能创建普通级别空间,,暂时这个校验去除,不然用户无法创建其他级别空间
//        if (SpaceLevelEnum.COMMON.getValue() != space.getSpaceLevel() && !userService.isAdmin(lgoinUser)) {
//            throw new BusinessException(ErrorCode.NO_AUTH_ERROR.getCode(), "无权限创建指定级别的空间");
//        }
        // 4.可控制同一个用户只能创建一个私有空间
        // 使用 String.intern()方法来获取一个字符串常量，并确保该字符串在字符串池中唯一。这种方式缺点:会导致数据不会及时释放,占用内存
//        String lock = String.valueOf(userId).intern();

        // 优化方案
        // 使用computeIfAbsent方法在lockMap中查找与userId关联的对象。
        // 如果lockMap中不存在该userId对应的键值对，则将传入的Lambda表达式的结果作为值与userId关联并存入map中。
        // Lambda表达式 (k) -> new Object() 的作用是：当给定的userId不存在于lockMap时，创建一个新的Object对象作为锁。
        // 最后，此方法返回与userId关联的对象（如果已经存在则直接返回已存在的对象；如果不存在，则返回新创建的对象）。
        // 这个对象可以用于同步块（synchronized block）以实现线程安全的操作。
        Object lock = lockMap.computeIfAbsent(userId, (k) -> new Object());

        synchronized (lock) {
            try {
                Long newSpaceId = transactionTemplate.execute(status -> {
                    // 判断当前用户是否已经创建过空间,,这里的逻辑可以自由调整,可以根据需求进行修改.
                    boolean exists = this.lambdaQuery()
                            .eq(Space::getUserId, userId)
                            .eq(Space::getSpaceType, space.getSpaceType())  // 每一种的空间类型只能创建一个
                            .exists();
                    // 如果存在，则不允许创建
                    ThrowUtils.throwIf(exists, ErrorCode.OPERATION_ERROR, "用户已经创建过空间");
                    boolean result = this.save(space);
                    ThrowUtils.throwIf(!result, ErrorCode.OPERATION_ERROR, "创建空间失败");

                    // 创建团队空间成功后,默认将创建者作为团队成员,添加到团队成员中,并设置为空间的管理员
                    if (space.getSpaceType() != null && SpaceTypeEnum.TEAM.getValue() == space.getSpaceType()) {
                        SpaceUser spaceUser = new SpaceUser();
                        spaceUser.setSpaceId(space.getId());
                        spaceUser.setUserId(userId);
                        spaceUser.setSpaceRole(SpaceRoleEnum.ADMIN.getValue());
                        spaceUser.setCreateUserId(userId);
                        spaceUser.setInviteStatus(InviteStatus.CONFIRM.getValue()); // 默认为已确认
                        boolean save = spaceUserService.save(spaceUser);
                        ThrowUtils.throwIf(!save, ErrorCode.OPERATION_ERROR, "将创建者添加到团队空间失败");
                    }

                    // 增加: 当空间创建完毕后,创建对应的分表, (仅对团队空间生效)
//                    dynamicShardingManager.createSpacePictureTable(space);

                    return space.getId();
                });
                return Optional.ofNullable(newSpaceId).orElse(-1L);
            } finally {
                lockMap.remove(userId);// 移除锁
            }

        }
    }

    @Override
    public QueryWrapper<Space> getQueryWrapper(SpaceQueryRequest spaceQueryRequest) {
        QueryWrapper<Space> queryWrapper = new QueryWrapper<>();
        if (spaceQueryRequest == null) {
            return queryWrapper;
        }
        Long id = spaceQueryRequest.getId();
        Long userId = spaceQueryRequest.getUserId();
        String spaceName = spaceQueryRequest.getSpaceName();
        Integer spaceLevel = spaceQueryRequest.getSpaceLevel();
        int pageNum = spaceQueryRequest.getPageNum();
        int pageSize = spaceQueryRequest.getPageSize();
        String sortField = spaceQueryRequest.getSortField();
        String sortOrder = spaceQueryRequest.getSortOrder();
        boolean asc = "asc".equals(sortOrder);
        Integer spaceType = spaceQueryRequest.getSpaceType();

        queryWrapper.eq(ObjUtil.isNotEmpty(id), "id", id);
        queryWrapper.eq(ObjUtil.isNotEmpty(userId), "userId", userId);
        queryWrapper.like(StrUtil.isNotBlank(spaceName), "spaceName", spaceName);
        queryWrapper.eq(ObjUtil.isNotEmpty(spaceLevel), "spaceLevel", spaceLevel);
        queryWrapper.eq(ObjUtil.isNotEmpty(spaceType), "spaceType", spaceType); // 增加空间类型搜索

        // 排序
        queryWrapper.orderBy(StrUtil.isNotEmpty(sortField), asc, sortField);
        return queryWrapper;
    }

    @Override
    public SpaceVO getSpaceVO(Space space, HttpServletRequest request) {
        SpaceVO spaceVO = SpaceVO.objToVo(space);
        if (spaceVO != null && spaceVO.getUserId() != null && spaceVO.getUserId() > 0) {
            // 获取用户信息
            User user = userService.getById(spaceVO.getUserId());
            UserVo userVo = userService.getUserVo(user);
            spaceVO.setUser(userVo);
        }
        return spaceVO;
    }

    @Override
    public Page<SpaceVO> getSpaceVOPage(Page<Space> spacePage, HttpServletRequest request) {
        if (spacePage == null) {
            return null;
        }
        List<Space> spaceList = spacePage.getRecords();
        Page<SpaceVO> spaceVOPage = new Page<>(spacePage.getCurrent(), spacePage.getSize(), spacePage.getTotal());
        if (CollUtil.isEmpty(spaceList)) {
            return spaceVOPage;
        }
        // 对象列表 => 封装对象列表
        List<SpaceVO> spaceVOList = spaceList.stream().map(SpaceVO::objToVo).collect(Collectors.toList());
        // 1. 关联查询用户信息 这里直接批量查询用户数据，降低数据库的访问次数
        Set<Long> userIdSet = spaceList.stream().map(Space::getUserId).collect(Collectors.toSet());
        Map<Long, List<User>> userIdUserListMap = userService.listByIds(userIdSet).stream().collect(Collectors.groupingBy(User::getId));
        // 2. 填充信息
        spaceVOList.forEach(spaceVO -> {
            Long userId = spaceVO.getUserId();
            User user = null;
            if (userIdUserListMap.containsKey(userId)) {
                user = userIdUserListMap.get(userId).get(0);
            }
            spaceVO.setUser(userService.getUserVo(user));
        });
        spaceVOPage.setRecords(spaceVOList);
        return spaceVOPage;
    }

    @Override
    public void validSpace(Space space, boolean add) {
        ThrowUtils.throwIf(space == null, ErrorCode.PARAMS_ERROR);
        // 从对象中取值
        String spaceName = space.getSpaceName();
        Integer spaceLevel = space.getSpaceLevel();
        Integer spaceType = space.getSpaceType();
        SpaceLevelEnum spaceLevelEnum = SpaceLevelEnum.getEnumByValue(spaceLevel);
        SpaceTypeEnum spaceTypeEnum = SpaceTypeEnum.getEnumByValue(spaceType);
        // 要创建
        if (add) {
            if (StrUtil.isBlank(spaceName)) {
                throw new BusinessException(ErrorCode.PARAMS_ERROR.getCode(), "空间名称不能为空");
            }
            if (spaceLevel == null) {
                throw new BusinessException(ErrorCode.PARAMS_ERROR.getCode(), "空间级别不能为空");
            }
            if (spaceType == null) {
                throw new BusinessException(ErrorCode.PARAMS_ERROR.getCode(), "空间类型不能为空");
            }
        }
        // 修改数据时，如果要改空间级别
        if (spaceLevel != null && spaceLevelEnum == null) {
            throw new BusinessException(ErrorCode.PARAMS_ERROR.getCode(), "空间级别不存在");
        }
        if (StrUtil.isNotBlank(spaceName) && spaceName.length() > 30) {
            throw new BusinessException(ErrorCode.PARAMS_ERROR.getCode(), "空间名称过长");
        }
        if (spaceType != null && spaceTypeEnum == null) {
            throw new BusinessException(ErrorCode.PARAMS_ERROR.getCode(), "空间类型不存在");
        }
    }

    @Override
    public void fillSpaceBySpaceLevel(Space space) {
        // 根据空间级别，自动填充限额
        SpaceLevelEnum spaceLevelEnum = SpaceLevelEnum.getEnumByValue(space.getSpaceLevel());
        if (spaceLevelEnum != null) {
            long maxSize = spaceLevelEnum.getMaxSize();
            if (space.getMaxSize() == null) {
                space.setMaxSize(maxSize);
            }
            long maxCount = spaceLevelEnum.getMaxCount();
            if (space.getMaxCount() == null) {
                space.setMaxCount(maxCount);
            }
        }
    }

    @Override
    public void checkSpaceAuth(Space space, User loginUser) {
        // 仅是管理员和创建者才允许操作空间
        if (!loginUser.getId().equals(space.getUserId()) && !userService.isAdmin(loginUser)) {
            throw new BusinessException(ErrorCode.NO_AUTH_ERROR.getCode(), "没有足够的权限");
        }
    }

}




